package org.codefilarete.stalactite.spring.repository.query;

import javax.sql.DataSource;
import java.util.IdentityHashMap;
import java.util.List;

import org.codefilarete.reflection.AccessorChain;
import org.codefilarete.stalactite.dsl.idpolicy.IdentifierPolicy;
import org.codefilarete.stalactite.engine.EntityPersister;
import org.codefilarete.stalactite.dsl.MappingEase;
import org.codefilarete.stalactite.engine.PersistenceContext;
import org.codefilarete.stalactite.engine.model.Color;
import org.codefilarete.stalactite.engine.model.Country;
import org.codefilarete.stalactite.engine.model.Language;
import org.codefilarete.stalactite.engine.model.Person;
import org.codefilarete.stalactite.engine.model.Republic;
import org.codefilarete.stalactite.engine.model.State;
import org.codefilarete.stalactite.engine.model.Timestamp;
import org.codefilarete.stalactite.engine.model.Vehicle;
import org.codefilarete.stalactite.engine.runtime.AdvancedEntityPersister;
import org.codefilarete.stalactite.id.Identifier;
import org.codefilarete.stalactite.query.model.JoinLink;
import org.codefilarete.stalactite.spring.repository.query.projection.ProjectionMappingFinder;
import org.codefilarete.stalactite.sql.Dialect;
import org.codefilarete.stalactite.sql.hsqldb.HSQLDBDialectBuilder;
import org.codefilarete.stalactite.sql.statement.binder.DefaultParameterBinders;
import org.codefilarete.stalactite.sql.statement.binder.LambdaParameterBinder;
import org.codefilarete.stalactite.sql.statement.binder.NullAwareParameterBinder;
import org.codefilarete.tool.collection.Arrays;
import org.codefilarete.tool.collection.Iterables;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.projection.SpelAwareProxyProjectionFactory;

import static org.assertj.core.api.Assertions.assertThat;
import static org.codefilarete.reflection.AccessorChain.fromMethodReference;
import static org.codefilarete.reflection.AccessorChain.fromMethodReferences;
import static org.codefilarete.reflection.AccessorDefinition.giveDefinition;
import static org.codefilarete.stalactite.dsl.MappingEase.entityBuilder;
import static org.codefilarete.stalactite.id.Identifier.LONG_TYPE;
import static org.codefilarete.stalactite.id.Identifier.identifierBinder;
import static org.codefilarete.stalactite.spring.repository.StalactiteRepositoryFactoryBean.asInternalPersister;
import static org.codefilarete.stalactite.sql.statement.binder.DefaultParameterBinders.INTEGER_PRIMITIVE_BINDER;
import static org.mockito.Mockito.mock;

class ProjectionMappingFinderTest {
	
	private static AdvancedEntityPersister<Republic, Identifier<Long>> entityPersister;
	
	@BeforeAll
	static void initPersister() {
		Dialect dialect = HSQLDBDialectBuilder.defaultHSQLDBDialect();
		dialect.getColumnBinderRegistry().register((Class) Identifier.class, identifierBinder(DefaultParameterBinders.LONG_PRIMITIVE_BINDER));
		dialect.getSqlTypeRegistry().put(Identifier.class, "int");
		
		dialect.getColumnBinderRegistry().register(Color.class, new NullAwareParameterBinder<>(new LambdaParameterBinder<>(INTEGER_PRIMITIVE_BINDER, Color::new, Color::getRgb)));
		dialect.getSqlTypeRegistry().put(Color.class, "int");
		
		PersistenceContext persistenceContext = new PersistenceContext(mock(DataSource.class), dialect);
		
		EntityPersister<Republic, Identifier<Long>> persister = entityBuilder(Republic.class, LONG_TYPE)
				.mapKey(Republic::getId, IdentifierPolicy.<Country, Identifier<Long>>alreadyAssigned(p -> p.getId().setPersisted(), p -> p.getId().isPersisted()))
				.map(Republic::getName)
				.map(Republic::getDescription)
				.map(Republic::isEuMember)
				.embed(Republic::getTimestamp, MappingEase.embeddableBuilder(Timestamp.class)
						.map(Timestamp::getCreationDate)
						.map(Timestamp::getModificationDate))
				.mapOneToOne(Republic::getPresident, entityBuilder(Person.class, LONG_TYPE)
						.mapKey(Person::getId, IdentifierPolicy.<Person, Identifier<Long>>alreadyAssigned(p -> p.getId().setPersisted(), p -> p.getId().isPersisted()))
						.map(Person::getName)
						.mapCollection(Person::getNicknames, String.class)
						.mapMap(Person::getPhoneNumbers, String.class, String.class)
						.mapOneToOne(Person::getVehicle, entityBuilder(Vehicle.class, LONG_TYPE)
								.mapKey(Vehicle::getId, IdentifierPolicy.<Vehicle, Identifier<Long>>alreadyAssigned(p -> p.getId().setPersisted(), p -> p.getId().isPersisted()))
								.map(Vehicle::getColor)))
				.mapOneToMany(Republic::getStates, entityBuilder(State.class, LONG_TYPE)
						.mapKey(State::getId, IdentifierPolicy.<State, Identifier<Long>>alreadyAssigned(p -> p.getId().setPersisted(), p -> p.getId().isPersisted()))
						.map(State::getName))
				.mapManyToMany(Republic::getLanguages, entityBuilder(Language.class, LONG_TYPE)
						.mapKey(Language::getId, IdentifierPolicy.<Language, Identifier<Long>>alreadyAssigned(p -> p.getId().setPersisted(), p -> p.getId().isPersisted()))
						.usingConstructor(Language::new, "id", "code")
						.map(Language::getCode).setByConstructor()
				)
				.build(persistenceContext);
		entityPersister = asInternalPersister(persister);
	}
	
	@Test
	void lookup_withClosedProjection() {
		SpelAwareProxyProjectionFactory projectionFactory = new SpelAwareProxyProjectionFactory();
		ProjectionMappingFinder<Republic> testInstance = new ProjectionMappingFinder<>(projectionFactory, entityPersister);
		IdentityHashMap<JoinLink<?, ?>, AccessorChain<Republic, ?>> selectablePaths = testInstance.lookup(NamesOnly.class);
		// we map the result to an ugly structure to be able to assert it easily because PropertyPath can't be created outside of Spring package
		// and JoinLinks can't be found easily.
		List<List<Object>> actual = Iterables.collectToList(selectablePaths.entrySet(), entry -> Arrays.asList(
				// column info
				entry.getKey().getOwner().getName(), entry.getKey().getExpression(), entry.getKey().getJavaType(),
				// propertyPath info
				giveDefinition(entry.getValue()))
		);
		assertThat(actual)
				.containsExactlyInAnyOrder(
						Arrays.asList("Person", "name", String.class, giveDefinition(fromMethodReferences(Country::getPresident, Person::getName))),
						Arrays.asList("Republic", "name", String.class, giveDefinition(fromMethodReference(Country::getName))),
						Arrays.asList("Vehicle", "color", Color.class, giveDefinition(fromMethodReferences(Country::getPresident, Person::getVehicle, Vehicle::getColor)))
				);
	}
	
	@Test
	void lookup_withOpenProjection() {
		SpelAwareProxyProjectionFactory projectionFactory = new SpelAwareProxyProjectionFactory();
		ProjectionMappingFinder<Republic> testInstance = new ProjectionMappingFinder<>(projectionFactory, entityPersister);
		IdentityHashMap<JoinLink<?, ?>, AccessorChain<Republic, ?>> selectablePaths = testInstance.lookup(NamesOnlyWithValue.class);
		// we map the result to an ugly structure to be able to assert it easily because PropertyPath can't be created outside of Spring package
		// and JoinLinks can't be found easily.
		List<List<Object>> actual = Iterables.collectToList(selectablePaths.entrySet(), entry -> Arrays.asList(
				// column info
				entry.getKey().getOwner().getName(), entry.getKey().getExpression(), entry.getKey().getJavaType(),
				// propertyPath info
				entry.getValue())
		);
		assertThat(actual).isEmpty();
	}
	
	interface NamesOnly {
		
		String getName();
		
		SimplePerson getPresident();
		
		interface SimplePerson {
			
			String getName();
			
			SimpleVehicle getVehicle();
			
		}
		
		interface SimpleVehicle {
			
			String getColor();
			
		}
	}
	
	// This class should create a query that retrieves the whole aggregate because it has a @Value annotation
	// Indeed, values required by the @Value annotations can't be known in advance, so the query must retrieve the whole aggregate
	interface NamesOnlyWithValue extends DerivedQueriesRepository.NamesOnly {
		
		@Value("#{target.president.name + '-' + target.president.id.delegate}")
		String getPresidentName();
	}
}
